Added an bugfix in the test