Simply if statement, fix sigma bug