fixed regression bug :/