Fix slight mess-up