a little fix (working now...)