Fixed minor bug because of merging

fixes small code mismatch