Fix some arithmetical errors

Please use `assertEquals`