Fix slight issue