fixing method missing for unit tests