Small fix #2

Really sorry.