quick change, rapidly fixed

[alexandrenoel20]

same code, just added an import for the code to look better when looked at