Minor bug in models