Attempt to fix code climate