Math fix, test still broken