Smalll fix on previous