Fixed "Weird gradient output"