Fix due to  #9