quick fix

Too late here. Made an error in my earlier commit.