Updating unit tests module