Small fix attempt