Bug temp fix. TODO