Fix stupid hard-code bug