fix this little bug