Bugfix

Little mistake was made.