Replace assertTrue with assertEquals where it made sensey